how to get user input and type casting in python